from flask import current_app, render_template, session, jsonify, request, g
from info.utils.response_code import RET
from . import index_blu
from ...models import User, News, Category
from info import constants

from info.utils.common import get_user_data


@index_blu.route('/')
@get_user_data
def index():
    """首页显示逻辑接口"""
    # ---------------1.用户登录成功，查询用户基本信息展示----------------
    """
    1.需要在用户请求首页的时候去数据库查询用户数据
    2.如果查询到用户数据，则传入模板中，进行渲染，再进行返回
    3.如果未查询到，返回用户数据为 None，模板代码中自行判断
    """
    user = g.user
    user_dict = user.to_dict() if user else None

    # ---------------2.查询新闻点击排行数据展示----------------
    try:
        news_list = News.query.order_by(News.clicks.desc()).limit(constants.CLICK_RANK_MAX_NEWS)
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.DBERR, errmsg="数据查询失败")
    # 新闻对象列表转化成字典列表
    news_dict_list = []
    """
    # 普通写法
    if news_list:
        for news in news_list:
            news_dict_list.append(news.to_dict())
    """
    # 三元运算表达式
    for news in news_list if news_list else []:
        news_dict_list.append(news.to_dict())

    # ---------------3.查询新闻分类数据展示----------------
    try:
        category_list = Category.query.all()
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.DBERR, errmsg="查询数据异常")
    category_dict_list = []
    for category in category_list if category_list else []:
        category_dict_list.append(category.to_dict())

    return render_template("news/index.html", data={
        "user_info": user_dict,
        "news_list": news_dict_list,
        "category_list": category_dict_list
    })


# get请求地址：/news_list?cid=1&p=1&per_page=10
@index_blu.route('/news_list')
def get_news_list():
    """获取首页新闻列表数据"""
    """
    1.获取参数（args）
        1.1 cid:分类id，p：当前页码，默认值：1表示第一页数据，per_page:每一页多少条数据，默认值：10
    2.参数校验
        2.1 cid非空判断
        2.2 将数据int强制类型转换
    3.逻辑处理
        3.1 根据cid作为查询条件，新闻的时间降序排序,进行分页查询
        3.2 将新闻对象列表转换成字典列表
    4.返回值
    """
    # 1.1 cid:分类id，p：当前页码，默认值：1表示第一页数据，per_page:每一页多少条数据，默认值：10
    params = request.args
    cid = params.get("cid")
    p = params.get("page", 1)
    per_page = params.get("per_page", constants.HOME_PAGE_MAX_NEWS)

    # 2.1 cid非空判断
    if not cid:
        return jsonify(errno=RET.PARAMERR, errmsg="cid值不存在")
    # 2.2 将数据int强制类型转换
    try:
        cid = int(cid)
        p = int(p)
        per_page = int(per_page)
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.PARAMERR, errmsg="参数类型错误")
    # 3.1 根据cid作为查询条件，新闻的时间降序排序,进行分页查询
    """
    # 查询的是最新的分类新闻数据，最新新闻查询只需要根据时间降序排序即可
    if cid == 1:
        paginate = News.query.order_by(News.create_time.desc()).paginate(p, per_page, False)
        # 获取当前所有数据
        news_list = paginate.items
        # 获取当前页码
        current_page = paginate.page
        # 获取总页数
        total_page = paginate.pages
    # 查询其他分类新闻需要加上：News.category_id == cid
    else:
        paginate = News.query.filter(News.category_id == cid).order_by(News.create_time.desc()).paginate(p, per_page, False)
        # 获取当前所有数据
        news_list = paginate.items
        # 获取当前页码
        current_page = paginate.page
        # 获取总页数
        total_page = paginate.pages
    """
    # 查询条件的列表
    filter_list = []
    filter_list.append(News.category_id == cid) if cid != 1 else None
    try:
        paginate = News.query.filter(*filter_list).order_by(News.create_time.desc()).paginate(p, per_page, False)
        # 获取当前所有数据
        news_list = paginate.items
        # 获取当前页码
        current_page = paginate.page
        # 获取总页数
        total_page = paginate.pages
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.DBERR, errmsg="数据查询异常")

    news_dict_list = []
    for news in news_list if news_list else []:
        news_dict_list.append(news.to_dict())

    return jsonify(errno=RET.OK, errmsg="查询新闻列表成功", data={
        "current_page": current_page,
        "total_page": total_page,
        "news_list": news_dict_list
    })


@index_blu.route('/favicon.ico')
def get_favicon():
    # 自定义视图函数，访问网站图标，send_static_file 是系统访问静态文件所调用的方法
    return current_app.send_static_file('news/favicon.ico')
